from train import solver
import torch
from SINet.sinet import SINet
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print(device)

SINet_model = SINet(64)
SI_solver = solver(SINet_model, device, epoch=30, lr=1e-4)

# 训练
SI_solver.train("./log/")
SI_solver.test("./log/")